package com.yls.except.util.imgfile;

import com.yls.except.util.ImgProperties;
import com.yls.except.util.UtilDate;

import java.io.*;
import java.net.HttpURLConnection;
import java.net.URL;
import java.util.UUID;

/**
 * Created by xufan on 2016/9/29.
 */
public class ImgFileHandle {
    /**
     * 测试
     * @param args
     */
    public static void main(String[] args) {
        String url = "https://mp.weixin.qq.com/cgi-bin/showqrcode?ticket=gQEJ8TwAAAAAAAAAAS5odHRwOi8vd2VpeGluLnFxLmNvbS9xLzAya2JudWhyWF9jRjAxMDAwME0wN3MAAgRCpK9ZAwQAAAAA";
        uploadimg(url, "D:\\", "img");
    }

    public static String uploadimg(String imgurl, String path, String src) {
        byte[] btImg = getImageFromNetByUrl(imgurl);
        if(null != btImg && btImg.length > 0){
            String fileName = UUID.randomUUID() + ".png";
            return writeImageToDisk(btImg, fileName, path, src);
        }else{
            System.out.println("没有从该连接获得内容");
            return null;
        }
    }
    /**
     * 将图片写入到磁盘
     * @param img 图片数据流
     * @param fileName 文件保存时的名称
     * @param src
     */
    public static String writeImageToDisk(byte[] img, String fileName, String path, String src){
        try {
            String date = UtilDate.getShijian();
            String filepath = path + date;
            File image = new File(filepath);
            if (!image.exists() && !image.isDirectory()) {
                image.mkdirs();
            }
            File file = new File(filepath + "/" + fileName);
            FileOutputStream fops = new FileOutputStream(file);
            fops.write(img);
            fops.flush();
            fops.close();

            //上传到阿里云
            InputStream in = new FileInputStream(filepath + "/" + fileName);
            path = OssUpload.UpImgToAlisOss(src + date  + fileName, in, "image/jpeg");
            return ImgProperties.getimg("filepath") + src + date  + fileName;
        } catch (Exception e) {
            e.printStackTrace();
        }
        return null;
    }
    /**
     * 根据地址获得数据的字节流
     * @param strUrl 网络连接地址
     * @return
     */
    public static byte[] getImageFromNetByUrl(String strUrl){
        try {
            URL url = new URL(strUrl);
            HttpURLConnection conn = (HttpURLConnection)url.openConnection();
            conn.setRequestMethod("GET");
            conn.setConnectTimeout(5 * 1000);
            InputStream inStream = conn.getInputStream();//通过输入流获取图片数据
            byte[] btImg = readInputStream(inStream);//得到图片的二进制数据
            return btImg;
        } catch (Exception e) {
            e.printStackTrace();
        }
        return null;
    }
    /**
     * 从输入流中获取数据
     * @param inStream 输入流
     * @return
     * @throws Exception
     */
    public static byte[] readInputStream(InputStream inStream) throws Exception{
        ByteArrayOutputStream outStream = new ByteArrayOutputStream();
        byte[] buffer = new byte[1024];
        int len = 0;
        while( (len=inStream.read(buffer)) != -1 ){
            outStream.write(buffer, 0, len);
        }
        inStream.close();
        return outStream.toByteArray();
    }
}
